package com.situ.mall.util;

import java.nio.charset.StandardCharsets;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;

public class MD5Util {

    /**
     * 对字符串进行 MD5 哈希
     * @param input 原始字符串
     * @return 32 位十六进制哈希字符串
     */
    public static String encrypt(String input) {
        try {
            MessageDigest md = MessageDigest.getInstance("MD5");
            byte[] bytes = md.digest(input.getBytes(StandardCharsets.UTF_8));
            return bytesToHex(bytes);
        } catch (NoSuchAlgorithmException e) {
            throw new RuntimeException("MD5 算法不存在", e);
        }
    }

    /**
     * 对字符串进行加盐的 MD5 哈希（仅用于增强简单性，不是强加密）
     * @param input 原始字符串
     * @param salt 盐值
     * @return 加盐后的哈希结果
     */
    public static String encryptWithSalt(String input, String salt) {
        return encrypt(input + salt);
    }

    /**
     * 字节数组转十六进制字符串
     */
    private static String bytesToHex(byte[] bytes) {
        StringBuilder hexString = new StringBuilder();
        for (byte b : bytes) {
            String hex = Integer.toHexString(0xff & b);
            if (hex.length() == 1) hexString.append('0');
            hexString.append(hex);
        }
        return hexString.toString();
    }

    // 简单测试
    public static void main(String[] args) {
        String text = "123456";
        String salt = "situ";

        System.out.println("原文: " + text);
        System.out.println("MD5: " + encrypt(text));
        System.out.println("加盐后: " + encryptWithSalt(text, salt));
    }
}
